from langchain_community.llms import Tongyi
from langchain.prompts import ChatPromptTemplate
import os

os.environ["DASHSCOPE_API_KEY"] = 'sk-c44402d7a12c41299bb716af8d7e8bac'

system_prompt_content = """
    你是一个翻译专家，你可以把{input_language}翻译成{output_language}，你可以根据提供的语调适当调整翻译结果，你只能输出翻译内容，用### 翻译后内容 ###标记
"""

human_prompt_content = """
    文本
    ###
    {user_input}
    ###
    语调 {style}
"""

prompt_template = ChatPromptTemplate.from_messages(
    [
        ('system', system_prompt_content),
        ('human', human_prompt_content)
    ]
)

user_input = 'Hello, how are you?'

prompt = prompt_template.invoke({'input_language': '英文', 'output_language': '中文', 'user_input': user_input, 'style': '讽刺'})

llm = Tongyi()

response = llm.invoke(prompt)

print(response)